Calculate cube of a number in python by lambda function